package com.test.web.suite;

import com.test.common.randcode.GetCodeImage;
import com.test.common.randcode.RemoveInterference;
import com.test.common.randcode.VerificationCode;
import com.test.web.enums.FindTypeEnum;
import org.apache.commons.io.FileUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.openqa.selenium.*;
import sun.misc.BASE64Decoder;

import javax.imageio.ImageIO;
import javax.imageio.ImageReadParam;
import javax.imageio.ImageReader;
import javax.imageio.stream.ImageInputStream;
import java.awt.image.BufferedImage;
import java.io.*;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Date;
import java.util.Iterator;

/**
 * Created by Administrator on 2017/6/20.
 */

public class SuiteMethods {

    private static final Logger logger = LogManager.getLogger();

    public void loginTest(WebDriver driver){
        Method lastMethod=getMethod(3);
        if(null==lastMethod){
            return;
        }
        LoginLocation location=null;
        if(lastMethod.isAnnotationPresent(LoginLocation.class)){
            location = lastMethod.getAnnotation(LoginLocation.class);
        }else{
            logger.error("请使用注解@LoginLocation");
            new AnnotationException().annotationUndefinedException("请使用注解@LoginLocation");
        }
        if(!location.hasTest()){
            driver.findElement(getByObject(location.locationType()[0],location.locationKey()[0])).sendKeys(location.trueValue()[0]);
            driver.findElement(getByObject(location.locationType()[1],location.locationKey()[1])).sendKeys(location.trueValue()[1]);
            if(location.locationType().length>=5){
                if(location.checkIsWork()){
                    WebElement codeImg=driver.findElement(getByObject(location.locationType()[3],location.locationKey()[3]));
                    String imgName=null;
                    try {
                        driver.manage().window().maximize();
                        imgName=createElementImage(codeImg,((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE));
                    } catch (IOException e) {
                        e.printStackTrace();
                        return;
                    }
//                    JavascriptExecutor oJavaScriptExecutor = (JavascriptExecutor)driver;
//                    oJavaScriptExecutor.executeScript("window.open('" + codeImg.getAttribute("src") + "')");
                    //GetCodeImage.downloadImage("https://mpf.aichel.com/basicInfo/createValidateAction.do");
                    RemoveInterference.removeInterference(imgName);
                    String code=VerificationCode.getCode(imgName);
                    driver.findElement(getByObject(location.locationType()[4],location.locationKey()[4])).sendKeys(code);
                    try {
                        Thread.sleep(5000);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                    return;
                }
            }
        }
        if(location.locationType().length!=location.locationKey().length){
            logger.error("注解LoginLocation：locationType和locationKey参数不匹配");
            new AnnotationException().annotationParameterdException("注解LoginLocation：locationType和locationKey参数不匹配");
        }
        ///System.out.println(value1);
    }

    private Method getMethod(int i){
        try {
            Class superClass = Class.forName(Thread.currentThread().getStackTrace()[i].getClassName());
            return superClass.getMethod(Thread.currentThread().getStackTrace()[i].getMethodName());
        } catch (ClassNotFoundException e) {
            logger.error("套件测试代码异常，请联系管理员");
            e.printStackTrace();
            new CodeException().globleCodeException("套件测试代码异常，请联系管理员");
            return null;
        } catch (NoSuchMethodException e) {
            logger.error("请使用无参testNG方法");
            e.printStackTrace();
            new CodeException().testNgCodedException("请使用无参testNG方法");
            return null;
        }
    }

    private By getByObject(FindTypeEnum type,String key){
        By by=null;
        try {
            by= (By) By.class.getMethod(type.getType(),new Class[]{String.class}).invoke(By.class,key);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        } catch (NoSuchMethodException e) {
            e.printStackTrace();
        }finally {
            return by;
        }
    }

    public String createElementImage(WebElement webElement,File file) throws IOException {
        String filenName =null;
        FileInputStream is = null;
        ImageInputStream iis = null;
        try {
            Point location = webElement.getLocation();
            Dimension size = webElement.getSize();
            // 读取图片文件
            is = FileUtils.openInputStream(file);

            // 获取文件的后缀名
            String postFix = file.getName().substring(file.getName().lastIndexOf(".") + 1);
            /*
             * 返回包含所有当前已注册 ImageReader 的 Iterator，这些 ImageReader 声称能够解码指定格式。
             * 参数：formatName - 包含非正式格式名称 .（例如 "jpeg" 或 "tiff"）等 。
             */
            Iterator<ImageReader> it = ImageIO.getImageReadersByFormatName(postFix);

            ImageReader reader = it.next();
            // 获取图片流
            iis = ImageIO.createImageInputStream(is);

            /*
             * <p>iis:读取源.true:只向前搜索 </p>.将它标记为 ‘只向前搜索’。
             * 此设置意味着包含在输入源中的图像将只按顺序读取，可能允许 reader 避免缓存包含与以前已经读取的图像关联的数据的那些输入部分。
             */
            reader.setInput(iis, true);

            /*
             * <p>描述如何对流进行解码的类<p>.用于指定如何在输入时从 Java Image I/O
             * 框架的上下文中的流转换一幅图像或一组图像。用于特定图像格式的插件 将从其 ImageReader 实现的
             * getDefaultReadParam 方法中返回 ImageReadParam 的实例。
             */
            ImageReadParam param = reader.getDefaultReadParam();

            /*
             * 图片裁剪区域。Rectangle 指定了坐标空间中的一个区域，通过 Rectangle 对象
             * 的左上顶点的坐标（x，y）、宽度和高度可以定义这个区域。
             */
            java.awt.Rectangle rect = new java.awt.Rectangle(location.x, location.y, size.width, size.height);

            // 提供一个 BufferedImage，将其用作解码像素数据的目标。
            param.setSourceRegion(rect);
            /*
             * 使用所提供的 ImageReadParam 读取通过索引 imageIndex 指定的对象，并将 它作为一个完整的
             * BufferedImage 返回。
             */
            BufferedImage bi = reader.read(0, param);
            filenName =new Date().getTime()+".jpg";
            ImageIO.write(bi, postFix, new File( "validateCode/img/"+filenName));
        } finally {
            if (is != null)
                is.close();
            if (iis != null)
                iis.close();
            return filenName;
        }
    }

}
